package leetcode.hash;

import java.util.HashSet;
import java.util.Set;

/**
 * 202. 快乐数
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 * 示例：
 * <p>
 * 输入：19
 * 输出：true
 * 解释：
 * 1^2 + 9^2 = 82
 * 8^2 + 2^2 = 68
 * 6^2 + 8^2 = 100
 * 1^2 + 0^2 + 0^2 = 1
 * <p>
 * #思路
 */
public class _202 {
    public static void main(String[] args) {
        boolean happy = isHappy(19);
    }

    public static boolean isHappy(int n) {
        Set<Integer> set = new HashSet<>();
        int sum = 0;
        while (!set.contains(sum)) {
            set.add(sum);
            sum = getSum(n);
            n = sum;
        }
        return sum == 1;

    }

    public static int getSum(int n) {
        int sum = 0;

        while (n > 0) {
            // 最后一位上的数字
            int temp = n % 10;
            sum = sum + temp * temp;
            n = n / 10;
        }
        return sum;
    }
}
